package com.atguigu.gmall.common.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * 当前配置类的作用：提供两个回调方法
 * 回顾：『回调』是指我们声明的方法，被系统调用
 *      confirm()方法：确认消息是否发送到交换机，通过传入的参数告诉我们发送是否成功
 *      returnedMessage()方法：消息没有发送到队列时，回调这个方法
 *
 * ConfirmCallback 接口中需要实现：confirm()方法
 * ReturnCallback 接口中需要实现：returnedMessage()方法
 *
 * 当前类的初始化操作：
 *      目标：为了给 RabbitTemplate 对象设置两个实例组件
 *          ConfirmCallback 实例：当前配置类的对象，也就是 this
 *          ReturnCallback 实例：当前配置类的对象，也就是 this
 *      调用方法：
 *          rabbitTemplate.setConfirmCallback(this)
 *          rabbitTemplate.setReturnCallback(this)
 *      初始化时机：当前类创建对象之后即可，使用 @PostConstructor 注解来对接这个时机
 *
 * @author: 封捷
 * @create-date: 2022/9/9 9:58
 */
@Configuration
@Slf4j
public class MQProducerAckConfig
        implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    // 要求1：修饰的方法不能有返回值
    // 要求2：修饰的方法不能是static
    // 注解修饰的方法执行的时机：创建对象之后，初始化之前
    @PostConstruct
    public void initRabbitTemplate() {
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 确认消息是否发送到了交换机
     * @param correlationData 相关数据
     * @param ack
     *      true：表示消息发送到了交换机
     *      false：表示选项没有发送到交换机
     * @param cause 如果没有发送到交换机，那么发送失败的原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack) {
            log.info("[配置类]消息成功发送到交换机！");
        } else {
            log.info("[配置类]消息发送到交换机失败！原因是：" + cause);
        }
    }

    /**
     * 在消息没有发送到队列的时候，会被回调。
     * 消息发送到交换机之后，根据路由键（routingKey）对接对应的队列。
     * 所以我们在找到交换机之后，往往会引用 routingKey的错误而找不到队列。
     * @param message 发送的消息
     * @param replyCode 当前状态描述应答码
     * @param replyText 当前状态描述
     * @param exchange 当前所在的交换机
     * @param routingKey 当前指定的路由键
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {

        // 消息的主体是一个字节数组
        byte[] messageBodyBytes = message.getBody();

        // 为了便于显示消息字符串，我们需要把字节数组传入 String 构造器
        String messageBody = new String(messageBodyBytes);

        log.info("消息主体: " + messageBody);
        log.info("应答码: " + replyCode);
        log.info("描述：" + replyText);
        log.info("消息使用的交换器 exchange : " + exchange);
        log.info("消息使用的路由键 routing : " + routingKey);
    }
}
